
export EVAL_ERTACACHE=0
export SAVE_ERROR=0
export CACHE_ADJUST=0
export EVAL_ERTACACHE=0
export TIMESTEP_ALIGNMENT=0
export DATA_TEST=2
export SKIP_LIST="[7, 9, 11, 13, 15, 16, 18, 19, 20, 21, 23, 24, 25, 26, 27, 28, 29, 31, 32, 33, 34, 35, 36, 37, 38, 39, 41, 42, 43, 44, 45, 46, 48]"
export PROMPT_PATH_LONGER="your_folder/ERTACache/VBench_full_info_longer.json"
export PROMPT_PATH="your_folder/ERTACache/VBench_full_info.json"
export CKPT_DIR="your_model_folder/THUDM/CogVideoX-2b"
export PROJECT_DOLDER="your_folder/ERTACache"
export PYTHONPATH=$PROJECT_DOLDER:$PYTHONPATH


############## original
export TIMESTEP_ALIGNMENT=0
export EVAL_ERTACACHE=0
export SAVE_ERROR=1
export CACHE_ADJUST=0
CUDA_VISIBLE_DEVICES=0 python3 ertacache_cogvideox.py \
    --ckpts_path $CKPT_DIR \
    --num_inference_steps 50 \
    --output_path ./sample/original \
    --height 480 \
    --width 720 \
    --fps 8 \
    --num_frames 49 \



############### time align
export TIMESTEP_ALIGNMENT=1
export EVAL_ERTACACHE=1
export SAVE_ERROR=1
export CACHE_ADJUST=0
CUDA_VISIBLE_DEVICES=0 python3 ertacache_cogvideox.py \
    --ckpts_path $CKPT_DIR \
    --num_inference_steps 50 \
    --output_path ./sample/ertacache_time_aligned \
    --height 480 \
    --width 720 \
    --fps 8 \
    --num_frames 49 \



############### time align + adjust
export EVAL_ERTACACHE=1
export SAVE_ERROR=0
export CACHE_ADJUST=1
export TIMESTEP_ALIGNMENT=1
export DATA_FOLDER_ORI="./sample/original/tensor"
export ADJUST_FOLDER="./sample/ertacache_time_aligned"
export DATA_FOLDER_CACHE="$ADJUST_FOLDER/ertacache_tensor"
python3 $PROJECT_DOLDER/adjust_utils/adjust_data_analysis.py
CUDA_VISIBLE_DEVICES=0 python3 ertacache_cogvideox.py \
    --ckpts_path $CKPT_DIR \
    --num_inference_steps 50 \
    --output_path ./sample/ertacache_timealign_adjust \
    --height 480 \
    --width 720 \
    --fps 8 \
    --num_frames 49 \
